DataFrames Per-Partition Counts in spark scala in Databricks